import os
from glob import glob
from setuptools import find_packages,setup

package_name = 'ball_collector'

setup(
    name=package_name,
    version='0.0.0',
    packages=[package_name],
    data_files=[
        ('share/ament_index/resource_index/packages',
            ['resource/' + package_name]),
        ('share/' + package_name, ['package.xml']),
        # 启动文件
        (os.path.join('share', package_name, 'launch'), glob('launch/*.launch.py')),
        # 配置文件
        (os.path.join('share', package_name, 'config'), glob('config/*.yaml')),
        # RViz配置
        (os.path.join('share', package_name, 'config/rviz'), glob('config/rviz/*.rviz')),
        # 模型目录
        (os.path.join('share', package_name, 'models'), []),
    ],
    install_requires=['setuptools'],
    zip_safe=True,
    maintainer='dudu',
    maintainer_email='2177527864@qq.com',
    description='Ball collection robot based on ROS2 and YOLO detection',
    license='Apache License 2.0',
    tests_require=['pytest'],
    entry_points={
        'console_scripts': [
            'detection_node = ball_collector.detection_node:main',
            'strategy_node = ball_collector.strategy_node:main',
            'hardware_interface = ball_collector.hardware_interface:main',
            'target_detection_node = ball_collector.target_detection_node:main',
        ],
    },
)